#ifndef RIVIAENTITYMANAGER_H
#define RIVIAENTITYMANAGER_H

#include <QHash>

class RIVIAImage;

namespace cv {
class Mat;
}

class RIVIAEntityManager
{
public:
    RIVIAEntityManager();
    virtual ~RIVIAEntityManager();

public:
    RIVIAImage* newImage(const cv::Mat &Mat);
    RIVIAImage* getCurImage();
    void deleteImageByID(int nImageID);

    void setCurImageID(int nImageID);

private:
    int m_nCurImageID;

    int m_nMaxImageID;
    QHash<int, RIVIAImage*>* m_pImagesHash;
};

#endif // RIVIAENTITYMANAGER_H
